python for loop vs while loop